package com.bite.common.utils;

import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.StringRedisTemplate;

import java.util.concurrent.TimeUnit;

@Slf4j
public class Redis {

    private StringRedisTemplate redisTemplate;

    private static final String REDIS_SPLIT = ":";

    private static final String REDIS_DEFAULT_PREFIX = "default";

    public Redis(StringRedisTemplate redisTemplate) {
        this.redisTemplate = redisTemplate;
    }

    /**
     * redis get
     */
    public String get(String key) {
        try {
            return key == null ? null : redisTemplate.opsForValue().get(key);
        } catch (Exception e) {
            log.error("redis get error, key:{}", key);
            return null;
        }
    }

    /**
     * hasKey
     */
    public boolean hasKey(String key) {
        try {
            return key == null ? false : redisTemplate.hasKey(key);
        } catch (Exception e) {
            log.error("redis hasKey error, key:{}", key);
            return false;
        }
    }

    /**
     * set Key
     */
    public boolean set(String key, String value) {
        try {
            // todo:参数校验
            redisTemplate.opsForValue().set(key, value);
            return true;
        } catch (Exception e) {
            log.error("redis set error, key:{}, value:{}, e:{}", key, value, e);
            return false;
        }
    }

    /**
     * set Key  添加过期时间，单位是秒
     */
    public boolean set(String key, String value, long timeout) {
        try {
            // todo:参数校验
            if (timeout > 0) {
                redisTemplate.opsForValue().set(key, value, timeout, TimeUnit.SECONDS);
            } else {
                set(key, value);
            }
            return true;
        } catch (Exception e) {
            log.error("redis set error, key:{}, value:{}", key, value);
            return false;
        }
    }

    public String buildKey(String prefix, String... args) {
        if (prefix == null) {
            prefix = REDIS_DEFAULT_PREFIX;
        }
        StringBuilder key = new StringBuilder();
        key.append(prefix);
        if (args != null) {
            for (String arg : args) {
                key.append(REDIS_SPLIT).append(arg);
            }
        }
        return key.toString();
    }
}
